#!/usr/bin/python
# -*- coding: utf-8 -*-
# @version        : 1.0
# @Create Time    : 2024/8/27 15:01
# @File           : plan
# @IDE            : PyCharm
# @desc           : 辅料计划查询参数

from datetime import datetime
from fastapi import Depends
from core.dependencies import Paging, QueryParams


class NeedPlanParams(QueryParams):
    """
    列表分页
    """

    def __init__(
            self,
            params: Paging = Depends(),
            product_type: str = None,
            station: str = None,
            status: str = None,
    ):
        super().__init__(params)
        self.product_type = product_type
        self.station = station
        self.status = status


class RecogizeNeedParams(QueryParams):
    """
    列表分页
    """

    def __init__(
            self,
            product_type: str,
            station: str,
            start: datetime,
            end: datetime,
    ):
        super().__init__()
        self.product_type = product_type
        self.station = station
        self.start = ('between', (start, end))
        